from src.bean.base_enum import BaseEnum


def entropy_add_suffix(v):
    return f"entropy_{v}"


class ColEntropyFieldEnum(BaseEnum):
    per_packet_req_body = (entropy_add_suffix("per_packet_req_body"), '熵值：单个包的请求body')
    per_packet_res_body = (entropy_add_suffix("per_packet_res_body"), '熵值：单个包的响应body')
    per_packet_all_body = (entropy_add_suffix("per_packet_all_body"), '熵值：单个包的请求+响应body')
    session_req_body = (entropy_add_suffix("session_req_body"), '熵值：session中所有包的请求body')
    session_res_body = (entropy_add_suffix("session_res_body"), '熵值：session中所有包的请求响应body')
    session_all_body = (entropy_add_suffix("session_all_body"), '熵值：session中所有包的请求+响应body')
    per_packet_uri = (entropy_add_suffix("per_packet_uri"), '熵值：每个包的uri,响应包为0')
    dns_domain = (entropy_add_suffix("dns_domain"), '熵值：dns域名')
